from flask import Blueprint, request, jsonify
from llm.MedAI.system_prompt_store import system_prompt_store

system_prompt_bp = Blueprint('system_prompt', __name__)

@system_prompt_bp.route('/system-prompt/<prompt_id>', methods=['GET'])
def get_system_prompt(prompt_id):
    """
    根据id获取system_prompt内容
    ---
    parameters:
      - name: prompt_id
        in: path
        type: string
        required: true
        description: 系统提示词ID
    responses:
      200:
        description: 返回system_prompt内容
        schema:
          type: object
          properties:
            prompt:
              type: string
      404:
        description: 未找到
    """
    if prompt_id == '1':
        return jsonify({'prompt': '这是一个测试的system prompt。'})
    prompt = system_prompt_store.get(prompt_id)
    if prompt is None:
        return jsonify({'error': 'Not found'}), 404
    return jsonify({'prompt': prompt})
